import os

# 创建基础文件夹, nnUNet/documentation/set_environment_variables.md
preprocessed_dir = "nnUNetFrame/DATASET/nnUNet_preprocessed"
raw_dir = "nnUNetFrame/DATASET/nnUNet_raw"
trained_dir = "nnUNetFrame/DATASET/nnUNet_results"
# 这个文件夹纯放文件，可以不使用
download_dir = "nnUNetFrame/DATASET/user_download"

os.makedirs(preprocessed_dir, exist_ok=True)
os.makedirs(raw_dir, exist_ok=True)
os.makedirs(trained_dir, exist_ok=True)
os.makedirs(download_dir, exist_ok=True)

# 示例数据集在 https://drive.google.com/drive/folders/1HqEgzS8BV2c7xYNrZdEAnrHk7osJJ--2
import shutil
# 解压移动
shutil.move("nnUNetFrame/DATASET/user_download/Task04_Hippocampus/Task04_Hippocampus", raw_dir)
# 修改文件夹名称
shutil.move(os.path.join(raw_dir, "Task04_Hippocampus"), os.path.join(raw_dir, "Dataset004_Hippocampus"))